package com.xuecheng.media.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.xuecheng.media.mapper.MediaFilesMapper;
import com.xuecheng.media.mapper.MediaProcessHistoryMapper;
import com.xuecheng.media.mapper.MediaProcessMapper;
import com.xuecheng.media.model.po.MediaFiles;
import com.xuecheng.media.model.po.MediaProcess;
import com.xuecheng.media.model.po.MediaProcessHistory;
import com.xuecheng.media.service.MediaProcessService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.time.LocalDateTime;
import java.util.List;

/**
 * <p>
 *  服务实现类
 * </p>
 *
 * @author htj
 */
@Slf4j
@Service
public class MediaProcessServiceImpl extends ServiceImpl<MediaProcessMapper, MediaProcess> implements MediaProcessService {
    @Autowired
    MediaProcessMapper mapper;

    @Autowired
    MediaProcessHistoryMapper mediaProcessHistoryMapper;

    @Autowired
    MediaFilesMapper mediaFilesMapper;

    /**
     * 根据分片参数获得处理任务列表
     * @param shardIndex
     * @param shardTotal
     * @param count
     * @return
     */
    @Override
    public List<MediaProcess> getMediaProcessList(int shardIndex, int shardTotal, int count) {
        return mapper.selectListByShardIndex(shardTotal,shardIndex,count);
    }

    /**
     * 乐观锁控制是否开启执行任务
     * 因为1234分片,如果12掉线,但是他们已经开始处理相应的任务,就设置乐观锁,只有执行成功的才能开始执行
     * 我觉得这里不应该放在事务中,否则如果事务有大量业务执行,会导致后面的线程无法更新数据库,进而一直等待
     * 乐观锁执行失败,可以尝试重试,例如状态数加一
     * @param id
     * @return
     */
    @Override
    public boolean startTask(long id) {
        int result = mapper.startTask(id);
        return result<=0?false:true;
    }

    /**
     * 更新任务执行状态
     * @param taskId  任务id
     * @param status 任务状态
     * @param fileId  文件id
     * @param url url
     * @param errorMsg 错误信息
     */
    @Override
    public void saveProcessFinishStatus(Long taskId, String status, String fileId, String url, String errorMsg) {
//        查询不到任务直接返回
        MediaProcess mediaProcess = mapper.selectById(taskId);
        if(mediaProcess == null){
            return;
        }

//        更新失败
//         更新失败状态 失败原因 失败次数
        if("3".equals(status)){
            LambdaQueryWrapper<MediaProcess> lambdaQueryWrapper = new LambdaQueryWrapper<>();
            lambdaQueryWrapper.eq(MediaProcess::getId,taskId);
            mediaProcess.setStatus(status);
            mediaProcess.setErrormsg(errorMsg);
            mediaProcess.setFailCount(mediaProcess.getFailCount()+1);
            mapper.update(mediaProcess,lambdaQueryWrapper);
            log.debug("更新任务处理状态失败!,任务信息:{}",mediaProcess);
            return;
        }
//        更新成功
//        更新mediaFile的url
        MediaFiles mediaFiles = mediaFilesMapper.selectById(fileId);
        if(mediaFiles != null){
            mediaFiles.setUrl(url);
            mediaFilesMapper.updateById(mediaFiles);
        }
//        更新任务状态 url
        mediaProcess.setUrl(url);
        mediaProcess.setStatus("2");
        mediaProcess.setFinishDate(LocalDateTime.now());
        mapper.updateById(mediaProcess);
//         插入任务历史表
        MediaProcessHistory mediaProcessHistory = new MediaProcessHistory();
        BeanUtils.copyProperties(mediaProcess,mediaProcessHistory);
        mediaProcessHistoryMapper.insert(mediaProcessHistory);
//        删除任务表记录
        mapper.deleteById(mediaProcess.getId());
    }

}
